// Copyright (c) 2022-present, Shanghai Yunxi Technology Co, Ltd. All rights reserved.
//
// This software (KWDB) is licensed under Mulan PSL v2.
// You can use this software according to the terms and conditions of the Mulan PSL v2.
// You may obtain a copy of Mulan PSL v2 at:
//          http://license.coscl.org.cn/MulanPSL2
// THIS SOFTWARE IS PROVIDED ON AN "AS IS" BASIS, WITHOUT WARRANTIES OF ANY KIND,
// EITHER EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO NON-INFRINGEMENT,
// MERCHANTABILITY OR FIT FOR A PARTICULAR PURPOSE.
// See the Mulan PSL v2 for more details.

package failpoint

import (
	"context"
	"errors"
	"gitee.com/kwbasedb/kwbase/pkg/replication/manager"
	"github.com/pingcap/failpoint"
)

func init() {
	manager.InjectTestReplicationPanic = injectTestReplicationPanic
	manager.InjectTestReplicationError = injectTestReplicationError
}

// panic at first time
func injectTestReplicationPanic(ctx context.Context, fpName string) {
	failpoint.Inject(fpName, func() {
		panic(fpName)
	})
}

var countInjectErr int

// return err at first time
func injectTestReplicationError(ctx context.Context, fpName string) error {
	var err error
	failpoint.Inject(fpName, func() {
		if countInjectErr == 0 {
			err = errors.New(fpName)
		}
		countInjectErr++
	})
	return err
}
